Skip to content

Tensorflow fixes - #20

Merged
lisavader merged 4 commits into
mainfrom
parse_early
Mar 27, 2026
Merged

Tensorflow fixes#20
lisavader merged 4 commits into
mainfrom
parse_early

Conversation

@lisavader

@lisavader lisavader commented Mar 27, 2026

Copy link
Copy Markdown
Collaborator

Some fixes to tensorflow imports and logging

  • TF import was moved to top of main script, after arg parsing, to ensure environment variables are set before TF initializes. This avoids repeated TF imports.
  • In order to avoid TF being loaded early via keras in model_loader.py, using lazy import in ResidualModelBase
  • TF C++ error messages are suppressed in --cpu-only mode (since these are almost always CUDA / other GPU messages, and any other errors are expected to give a Python level error)
  • The uv.lock was updated with regards to the latest TF dependency changes

@lisavader
lisavader requested a review from lhon as a code owner March 27, 2026 04:20
@lisavader
lisavader merged commit b162b14 into main Mar 27, 2026
4 checks passed
@lisavader
lisavader deleted the parse_early branch March 27, 2026 05:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant